Goto

Collaborating Authors

 learned uncertainty-aware


Learned Uncertainty-Aware (LUNA) Bases for Bayesian Regression using Multi-Headed Auxiliary Networks

Thakur, Sujay, Lorsung, Cooper, Yacoby, Yaniv, Doshi-Velez, Finale, Pan, Weiwei

arXiv.org Machine Learning

Neural Linear Models (NLM) are deep models that produce predictive uncertainty by learning features from the data and then performing Bayesian linear regression over these features. Despite their popularity, few works have focused on formally evaluating the predictive uncertainties of these models. In this work, we show that traditional training procedures for NLMs can drastically underestimate uncertainty in data-scarce regions. We identify the underlying reasons for this behavior and propose a novel training procedure for capturing useful predictive uncertainties.